/**
 * Collaborative Story Development Web Application (CSDApp) 
 * Copyright Anton Strack 2014
 *
 * This file is part of Collaborative Story Development Application (CSDApp).
 *
 * CSDApp is free software: you can redistribute it and/or modify it under the
 * terms of the GNU General Public License as published by the Free Software
 * Foundation, either version 3 of the License, or any later version.
 *
 * CSDApp is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
 * A PARTICULAR PURPOSE. See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * CSDApp. If not, see <http://www.gnu.org/licenses/>.
 *
 */
package csdwa;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * This class provides database access to the specific resource model. It
 * provides CRUD database functionality and it stores it's results as a list.
 * Additionally, through it's methods, it provides access to determine the table
 * name, and query markers that are set for the particular DAO and thus allows
 * other code a means of interacting with this DAO to execute more advanced
 * queries.
 *
 * @author Anton Strack
 */
public class NotificationDao extends DaoSupport implements CsdwaDao<Notification> {

    private List<Notification> beans;

    public NotificationDao(Connection conn) {
        super(conn);
        this.beans = new ArrayList<Notification>();
        this.restoreQueryMarkers();
    }

    /**
     * Select all the beans from the bean table
     *
     * @return the row count indicating if 0 or more beans were retrieved from
     * the database.
     */
    @Override
    public int selectBeans() {
        int result = 0;
        this.usingQueryMarkers = this.getCoreSelectUsingQueryMarkers();
        this.setMarker(QueryMarker.CORE_CONDITIONS, "1=1");
        try {
            ps = getConn().prepareStatement(this.buildQueryString(this.usingQueryMarkers, SELECT_QUERY));
            result = executeSelectQuery(ps);
        } catch (SQLException ex) {
            Logger.getLogger(PostDao.class.getName()).log(Level.SEVERE, null, ex);
        }
        return result;
    }

    /**
     * Select a single bean by it's id.
     *
     * @param id the id number of the bean to select.
     * @return 1 or 0, indicating a bean was retrieved or not.
     */
    @Override
    public int selectBeanById(int id) {
        int result = 0;
        this.usingQueryMarkers = this.getCoreSelectUsingQueryMarkers();
        this.setMarker(QueryMarker.CORE_CONDITIONS, "notificationmailbox.ID = ?");
        try {
            ps = getConn().prepareStatement(this.buildQueryString(usingQueryMarkers, SELECT_QUERY));
            this.ps.setInt(1, id);
            result = executeSelectQuery(ps);
        } catch (SQLException ex) {
            Logger.getLogger(PostDao.class.getName()).log(Level.SEVERE, null, ex);
        }
        return result;
    }

    /**
     * Select all the beans from the bean table matching the notificationTypeId
     *
     * @param id the id number of the bean to select.
     * @return 1 or 0, indicating a bean was retrieved or not.
     */
    public int selectBeansByNotificationTypeId(int notificationTypeId) {
        int result = 0;
        this.usingQueryMarkers = this.getCoreSelectUsingQueryMarkers();
        this.setMarker(QueryMarker.CORE_CONDITIONS, "notificationmailbox.NotificationTypeID = ?");
        try {
            ps = getConn().prepareStatement(this.buildQueryString(usingQueryMarkers, SELECT_QUERY));
            this.ps.setInt(1, notificationTypeId);
            result = executeSelectQuery(ps);
        } catch (SQLException ex) {
            Logger.getLogger(PostDao.class.getName()).log(Level.SEVERE, null, ex);
        }
        return result;
    }

    /**
     * Select all the beans from the bean table matching the userId
     *
     * @param id the id number of the bean to select.
     * @return 1 or 0, indicating a bean was retrieved or not.
     */
    public int selectBeansByUserId(int UserId) {
        int result = 0;
        this.usingQueryMarkers = this.getCoreSelectUsingQueryMarkers();
        this.setMarker(QueryMarker.CORE_CONDITIONS, "notificationmailbox.UserID = ?");
        try {
            ps = getConn().prepareStatement(this.buildQueryString(usingQueryMarkers, SELECT_QUERY));
            this.ps.setInt(1, UserId);
            result = executeSelectQuery(ps);
        } catch (SQLException ex) {
            Logger.getLogger(PostDao.class.getName()).log(Level.SEVERE, null, ex);
        }
        return result;
    }

    /**
     * Select all the beans from the bean table matching the userId and filtered
     * by the stateId
     *
     * @param userId the id number of the user to filter the bean by
     * @param stateId the state id to filter the notification on.
     * @param negate if true the apply the state filter as Not and thus select
     * all not equal to the stateId
     * @return 1 or 0, indicating a bean was retrieved or not.
     */
    public int selectBeansByUserId(int userId, int stateId, boolean negate) {
        int result = 0;
        this.usingQueryMarkers = this.getCoreSelectUsingQueryMarkers();
        if (negate == true) {
            this.setMarker(QueryMarker.CORE_CONDITIONS, "notificationmailbox.UserID = ? AND notificationmailbox.StateID <> ?");
        } else {
            this.setMarker(QueryMarker.CORE_CONDITIONS, "notificationmailbox.UserID = ? AND notificationmailbox.StateID = ?");
        }

        try {
            ps = getConn().prepareStatement(this.buildQueryString(usingQueryMarkers, SELECT_QUERY));
            this.ps.setInt(1, userId);
            this.ps.setInt(2, stateId);
            result = executeSelectQuery(ps);
        } catch (SQLException ex) {
            Logger.getLogger(PostDao.class.getName()).log(Level.SEVERE, null, ex);
        }
        return result;
    }

    /**
     * Clear the bean list so it doesn't contain any beans.
     */
    @Override
    public void clearDaoBeans() {
        this.beans.clear();
    }

    /**
     * Get the bean stored in the dao by it's primary id.
     *
     * @param id the id of the bean to select.
     * @return the bean if it exists in the dao or null if it doesn't
     */
    @Override
    public Notification getBeanById(int id) {
        Notification bean = null;
        if (this.beans.isEmpty() == false) {
            for (Notification b : this.beans) {
                if (b.getId() == id) {
                    bean = b;
                    break;
                }
            }
        }
        return bean;
    }

    /**
     * Save a bean to the database
     *
     * @param bean the bean to insert. The bean id is not required and won't be
     * used. All other bean values are required.
     * @return the auto-generated id key for the new bean row or 0 on failure.
     */
    @Override
    public int insertBean(Notification bean) {
        int result = 0;
        ResultSet rs;
        this.usingQueryMarkers = new ArrayList<QueryMarker>();
        this.usingQueryMarkers.add(QueryMarker.CORE_COLUMNS);
        this.usingQueryMarkers.add(QueryMarker.CORE_TABLE);
        this.usingQueryMarkers.add(QueryMarker.INSERT_VALUES);
        /*
         * HSQLDB cannot properly handle insert statements that have table identifiesr in the form of table.columnName so we need to remove the table name for inserts.
         */
        this.queryMarkers.put(QueryMarker.CORE_COLUMNS, this.prepareInsertColumns(this.getMarker(QueryMarker.CORE_COLUMNS)));
        try {
            this.ps = this.getConn().prepareStatement(this.buildQueryString(usingQueryMarkers, INSERT_QUERY), Statement.RETURN_GENERATED_KEYS);
            this.ps.setNull(1, java.sql.Types.NULL);
            this.ps.setInt(2, bean.getResourceTypeId());
            this.ps.setInt(3, bean.getUserId());
            this.ps.setInt(4, bean.getNotificationTypeId());
            this.ps.setInt(5, bean.getResourceId());
            this.ps.setInt(6, bean.getStateId());
            this.ps.setInt(7, bean.getEventCreatorId());
            this.ps.setTimestamp(8, bean.getNotificationDate());
            this.ps.setString(9, bean.getResourceName());
            result = this.ps.executeUpdate();
            rs = ps.getGeneratedKeys();
            if (result != 0 && rs.next()) {
                result = rs.getInt(1); // get the auto-generated key from the successful insert operation.
            }
        } catch (SQLException ex) {
            Logger.getLogger(NotificationDao.class.getName()).log(Level.SEVERE, null, ex);
        }
        return result;
    }

    /**
     * Update a bean in the database as well as the bean in the dao's beans map.
     *
     * @param bean the bean to update.
     * @return 1 or 0 indicating if the bean was updated or not.
     */
    @Override
    public int updateBean(Notification bean) {
        int result = 0;
        this.usingQueryMarkers = new ArrayList<QueryMarker>();
        this.usingQueryMarkers.add(QueryMarker.CORE_CONDITIONS);
        this.usingQueryMarkers.add(QueryMarker.CORE_TABLE);
        this.usingQueryMarkers.add(QueryMarker.UPDATE_COLUMNS_VALUES);
        String[] cols = this.splitStringByCommas(this.COLUMNS_TO_UPDATE);
        this.setMarker(QueryMarker.UPDATE_COLUMNS_VALUES, this.prepareUpdateColumns(cols));
        this.setMarker(QueryMarker.CORE_CONDITIONS, "ID = ?");
        try {
            this.ps = getConn().prepareStatement(this.buildQueryString(usingQueryMarkers, UPDATE_QUERY));
            int index = 1;
            for (String col : cols) {
                if ("ResourceName".equalsIgnoreCase(col)) {
                    this.ps.setString(index, bean.getResourceName());
                    index++;
                } else if ("ResourceTypeId".equalsIgnoreCase(col)) {
                    this.ps.setInt(index, bean.getResourceTypeId());
                    index++;
                } else if ("StateID".equalsIgnoreCase(col)) {
                    this.ps.setInt(index, bean.getStateId());
                    index++;
                } else if ("EventCreatorID".equalsIgnoreCase(col)) {
                    this.ps.setInt(index, bean.getEventCreatorId());
                    index++;
                } else if ("UserID".equalsIgnoreCase(col)) {
                    this.ps.setInt(index, bean.getUserId());
                    index++;
                } else if ("NotificationDate".equalsIgnoreCase(col)) {
                    this.ps.setTimestamp(index, bean.getNotificationDate());
                    index++;
                } else if ("ResourceID".equalsIgnoreCase(col)) {
                    this.ps.setInt(index, bean.getResourceId());
                    index++;
                }
            }
            this.ps.setInt(index, bean.getId()); // where column condition
            result = this.ps.executeUpdate();
        } catch (SQLException ex) {
            Logger.getLogger(ResourceTypeDao.class.getName()).log(Level.SEVERE, null, ex);
        }
        return result;
    }

    /**
     * Deletes the bean from the database
     *
     * @param bean the bean to delete
     * @return 1 or 0 indicating if the bean was deleted or not.
     */
    @Override
    public int deleteBean(Notification bean) {
        int result = 0;
        this.usingQueryMarkers = new ArrayList<QueryMarker>();
        this.usingQueryMarkers.add(QueryMarker.CORE_CONDITIONS);
        this.usingQueryMarkers.add(QueryMarker.CORE_TABLE);
        this.setMarker(QueryMarker.CORE_CONDITIONS, "ID = ?");
        try {
            ps = getConn().prepareStatement(this.buildQueryString(usingQueryMarkers, DELETE_QUERY));
            this.ps.setInt(1, bean.getId());
            result = this.ps.executeUpdate();
        } catch (SQLException ex) {
            Logger.getLogger(NotificationDao.class.getName()).log(Level.SEVERE, null, ex);
        }
        return result;
    }

    /**
     * Reset the query markers to their default values for the dao.
     */
    @Override
    public void restoreQueryMarkers() {
        this.setMarker(QueryMarker.CORE_TABLE, "notificationmailbox");
        this.setMarker(QueryMarker.CORE_COLUMNS, "notificationmailbox.ID, notificationmailbox.ResourceTypeID, notificationmailbox.UserID, notificationmailbox.NotificationTypeID, notificationmailbox.ResourceID, notificationmailbox.StateID, notificationmailbox.EventCreatorID, notificationmailbox.NotificationDate, notificationmailbox.ResourceName");
        this.setMarker(QueryMarker.CORE_CONDITIONS, "notificationmailbox.ID = ?");
        this.setMarker(QueryMarker.ORDER_BY_COLUMN, "notificationmailbox.NotificationDate");
        this.setMarker(QueryMarker.ORDER_BY_MODE, "DESC");
        this.setMarker(QueryMarker.LIMIT, "");
        this.setMarker(QueryMarker.INSERT_VALUES, "?, ?, ?, ?, ?, ?, ?, ?, ?");
        this.setMarker(QueryMarker.UPDATE_COLUMNS_VALUES, "");
        this.setMarker(QueryMarker.PRIMARY_KEY, "ID");
        // special type of derived values
        this.setMarker(QueryMarker.RELATED_COLUMNS, ", crUser.Username AS EventCreatorName, rUser.Username AS UserName, state.name AS StateName, resourcetype.name AS ResourceTypeName, notificationtype.Name AS NotificationTypeName");
        this.setMarker(QueryMarker.RELATED_CONDITIONS, " AND notificationmailbox.UserID = rUser.ID AND notificationmailbox.EventCreatorID = crUser.ID AND notificationmailbox.StateID = state.ID AND notificationmailbox.NotificationTypeID = notificationtype.ID AND notificationmailbox.ResourceTypeID = resourcetype.ID");
        this.setMarker(QueryMarker.RELATED_TABLES, ", useraccount AS crUser, useraccount AS rUser, state, resourcetype, notificationtype");
        this.COLUMNS_TO_UPDATE = "";
    }

    /**
     * Execute the select query and place the results in beans list.
     *
     * @param ps the fully prepared PreparedStatement
     * @return the row count from the select statement
     */
    @Override
    protected int executeSelectQuery(PreparedStatement ps) {
        int result = 0;
        ResultSet rs;
        try {
            rs = ps.executeQuery();
            while (rs.next()) {
                Notification bean = ResourceFactory.makeNotification();
                bean.setId(rs.getInt("notificationmailbox.ID"));//
                bean.setNotificationDate(rs.getTimestamp("notificationmailbox.NotificationDate"));
                bean.setResourceTypeId(rs.getInt("notificationmailbox.ResourceTypeID"));
                bean.setResourceId(rs.getInt("notificationmailbox.ResourceID"));
                bean.setStateId(rs.getInt("notificationmailbox.StateID"));
                bean.setNotificationTypeId(rs.getInt("notificationmailbox.NotificationTypeID"));
                bean.setEventCreatorId(rs.getInt("notificationmailbox.EventCreatorID"));
                bean.setUserId(rs.getInt("notificationmailbox.UserID"));
                bean.setResourceName(rs.getString("notificationmailbox.ResourceName"));
                if (this.USE_RELATED_MARKERS == true) {
                    bean.setDerivedValue("UserName", rs.getString("UserName"));
                    bean.setDerivedValue("EventCreatorName", rs.getString("EventCreatorName"));
                    bean.setDerivedValue("StateName", rs.getString("StateName"));
                    bean.setDerivedValue("ResourceTypeName", rs.getString("ResourceTypeName"));
                    bean.setDerivedValue("NotificationTypeName", rs.getString("NotificationTypeName"));
                }
                this.processDerivedValues(bean, rs);
                this.beans.add(bean);
                result++;
            }
        } catch (SQLException ex) {
            Logger.getLogger(ActionDao.class.getName()).log(Level.SEVERE, null, ex);
        }


        return result;
    }

    /**
     * Get a list of beans
     *
     * @return a list of bean objects
     */
    @Override
    public List<Notification> getBeans() {
        return new ArrayList<Notification>(this.beans);
    }

    /**
     * Get a single bean.
     *
     * @return a single bean in the bean dao.
     */
    @Override
    public Notification getBean() {
        if (this.beans.isEmpty()) {
            return null;
        }
        return this.beans.get(0);
    }

    /**
     * Add a bean in the dao's bean list.
     *
     * @param bean the bean to add or update
     */
    @Override
    public void addBeanToDao(Notification bean) {
        this.beans.add(bean);
    }
}
